#!/usr/bin/env python

from nbdt.hierarchy import generate_hierarchy, test_hierarchy, generate_hierarchy_vis
from nbdt.graph import get_parser
from nbdt.thirdparty.wn import maybe_install_wordnet


def main():

    parser = get_parser()
    args = parser.parse_args()

    if not args.path:
        maybe_install_wordnet()
        generate_hierarchy(**vars(args))
        test_hierarchy(args)
    generate_hierarchy_vis(args)


if __name__ == "__main__":
    main()
